/*
 * Copyright (c) 2019  STMicroelectronics – All rights reserved
 * The STMicroelectronics corporate logo is a trademark of STMicroelectronics
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 * - Redistributions of source code must retain the above copyright notice, this list of conditions
 *   and the following disclaimer.
 *
 * - Redistributions in binary form must reproduce the above copyright notice, this list of
 *   conditions and the following disclaimer in the documentation and/or other materials provided
 *   with the distribution.
 *
 * - Neither the name nor trademarks of STMicroelectronics International N.V. nor any other
 *   STMicroelectronics company nor the names of its contributors may be used to endorse or
 *   promote products derived from this software without specific prior written permission.
 *
 * - All of the icons, pictures, logos and other images that are provided with the source code
 *   in a directory whose title begins with st_images may only be used for internal purposes and
 *   shall not be redistributed to any third party or modified in any way.
 *
 * - Any redistributions in binary form shall not include the capability to display any of the
 *   icons, pictures, logos and other images that are provided with the source code in a directory
 *   whose title begins with st_images.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
 * OF SUCH DAMAGE.
 */
package com.st.BlueMS.demos.plot

import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.ViewModel
import com.st.BlueSTSDK.Feature
import com.st.BlueSTSDK.Features.FeatureProximity
import com.st.BlueSTSDK.Node

internal data class PlotEntry(val x:Long,
                              val y:FloatArray)

internal class PlotDataViewModel()  : ViewModel(){

    lateinit var node : Node

    private var _lastPlotData = MutableLiveData<PlotEntry?>()
    val lastPlotData:LiveData<PlotEntry?>
    get() = _lastPlotData

    private val _isPlotting = MutableLiveData<Boolean>(false)
    val isPlotting:LiveData<Boolean>
    get() = _isPlotting

    private var _lastDataDescription = MutableLiveData<String>()

    val lastDataDescription:LiveData<String>
    get() = _lastDataDescription

    private var mFirstNotificationTimeStamp:Long = 0
    private var mCurrentPlottingFeature:Feature? = null

    private  var mNotificationEnabled = false

    private fun filterSample(f:Feature, sample:Feature.Sample): Feature.Sample{
        if( f is FeatureProximity){
            if( FeatureProximity.isOutOfRangeDistance(sample)) {
                sample.data[0] = 0
            }
        }
        return sample
    }

    private val mFeatureListener = Feature.FeatureListener{ f,sample ->
        //remove the starting time to avoid to have enormous x values
        val plotEntry = filterSample(f,sample).toPlotEntry(xOffset = mFirstNotificationTimeStamp)
        val lastX = _lastPlotData.value?.x ?: Long.MIN_VALUE
        if(plotEntry.x < lastX)
            return@FeatureListener
        _lastPlotData.postValue(plotEntry)
        _lastDataDescription.postValue(f.toStringToPlot())
    }

    fun startPlotFeature(f:Feature){
        stopPlot()
        mFirstNotificationTimeStamp = System.currentTimeMillis()
        f.addFeatureListener(mFeatureListener)
        //f.enableNotification()
        _isPlotting.value=true
        mCurrentPlottingFeature = f
        if(!mNotificationEnabled) {
            notificationStartStop()
        }
    }

    fun stopPlot(){
        mCurrentPlottingFeature?.apply {
            removeFeatureListener(mFeatureListener)
            if(mNotificationEnabled) {
                notificationStartStop()
            }
        }
        _isPlotting.value=false
        _lastPlotData.postValue(null)
        mCurrentPlottingFeature = null
    }

    fun onStartStopButtonPressed(selectedFeature:Feature){
        if(_isPlotting.value == true){
            stopPlot()
        }else{
            startPlotFeature(selectedFeature)
        }
    }

    fun onResumePauseButtonPressed(){
        notificationStartStop()
    }

    private fun notificationStartStop() {
        mCurrentPlottingFeature?.apply {
            mNotificationEnabled = if(mNotificationEnabled) {
                disableNotification()
                false
            } else {
                this.enableNotification()
                true
            }
        }
    }
}

private fun Feature.Sample.toPlotEntry(xOffset:Long): PlotEntry{
    val yData = FloatArray(data.size) { i -> data[i].toFloat() }.filterIndexed { index, _ -> dataDesc[index].plotIt  }.toFloatArray()
    return PlotEntry(notificationTime-xOffset,yData)
}